package com.wiloon.java8x.thread.threadPool;

/**
 * Created with IntelliJ IDEA.
 * User: wiloon
 * Date: 8/26/12
 * Time: 9:58 AM
 */

import java.util.LinkedList;

/**
 * @author sunnylocus
 * @project LocationGateway
 * @verson 1.0.0
 * @date Aug 2, 2008
 * @jdk 1.4.2
 */

public class ThreadPool extends ThreadGroup {
    private boolean isClosed = false;  //线程池是否关闭
    private LinkedList workQueue;      //工作队列
    private static int threadPoolID = 1;  //线程池的id

    public ThreadPool(int poolSize) {  //poolSize 表示线程池中的工作线程的数量
        super(threadPoolID + "");      //指定ThreadGroup的名称
        System.out.println("thread pool constructor");

        setDaemon(true);               //继承到的方法，设置是否守护线程池
        workQueue = new LinkedList();  //创建工作队列
        for (int i = 0; i < poolSize; i++) {
            new WorkThread(i).start();   //创建并启动工作线程,线程池数量是多少就创建多少个工作线程
            System.out.println("work thread " + i + " call start");

        }
    }

    /**
     * 向工作队列中加入一个新任务,由工作线程去执行该任务
     */
    public synchronized void execute(Runnable task) {
        System.out.println("thread pool.execute");

        if (isClosed) {
            throw new IllegalStateException();
        }
        if (task != null) {
            workQueue.add(task);//向队列中加入一个任务
            notify();             //唤醒一个正在getTask()方法中待任务的工作线程
            System.out.println("thread pool.work queue.add");

        }
    }

    /**
     * 从工作队列中取出一个任务,工作线程会调用此方法
     */
    private synchronized Runnable getTask(int threadid) throws InterruptedException {
        while (workQueue.size() == 0) {
            if (isClosed) return null;
            System.out.println("工作线程" + threadid + "等待任务...");
            wait();                //如果工作队列中没有任务,就等待任务
        }
        System.out.println("工作线程" + threadid + "开始执行任务...");
        return (Runnable) workQueue.removeFirst(); //反回队列中第一个元素,并从队列中删除
    }

    /**
     * 关闭线程池
     */
    public synchronized void closePool() {
        System.out.println("thread pool.close pool");

        if (!isClosed) {
            waitFinish();        //等待工作线程执行完毕
            isClosed = true;
            workQueue.clear();  //清空工作队列
            interrupt();         //中断线程池中的所有的工作线程,此方法继承自ThreadGroup类
            System.out.println("thread pool.interrupt");

        }
    }

    /**
     * 等待工作线程把所有任务执行完毕
     */
    public void waitFinish() {
        System.out.println("thread pool.wait finish");

        synchronized (this) {
            isClosed = true;
            notifyAll();            //唤醒所有还在getTask()方法中等待任务的工作线程
        }
        Thread[] threads = new Thread[activeCount()]; //activeCount() 返回该线程组中活动线程的估计值。
        int count = enumerate(threads); //enumerate()方法继承自ThreadGroup类，根据活动线程的估计值获得线程组中当前所有活动的工作线程
        for (int i = 0; i < count; i++) { //等待所有工作线程结束
            try {
                threads[i].join();    //等待工作线程结束
            } catch (InterruptedException ex) {
                ex.printStackTrace();
            }
        }
    }

    /**
     * 内部类,工作线程,负责从工作队列中取出任务,并执行
     *
     * @author sunnylocus
     */
    private class WorkThread extends Thread {
        private int id;

        public WorkThread(int id) {
            //父类构造方法,将线程加入到当前ThreadPool线程组中
            super(ThreadPool.this, id + "");
            this.id = id;
        }

        public void run() {
            while (!isInterrupted()) {  //isInterrupted()方法继承自Thread类，判断线程是否被中断
                Runnable task = null;
                try {
                    task = getTask(id);        //取出任务
                } catch (InterruptedException ex) {
                    ex.printStackTrace();
                }
                //如果getTask()返回null或者线程执行getTask()时被中断，则结束此线程
                if (task == null) return;

                try {
                    task.run();  //运行任务
                } catch (Throwable t) {
                    t.printStackTrace();
                }
            }//  end while
        }//  end run
    }// end workThread
}